#train_test_split is now in model_selection. Just type: from sklearn.model_selection import train_test_split #it should work